"""
数据库模块
提供数据库连接、会话管理和模型定义
"""

from .connection import get_db, engine, AsyncSessionLocal
from .models.base import Base
from .models.scan import ScanTask, ScanResult, ScanRecord, AIAnalysisResult
# 临时注释掉file模块导入，避免启动失败
# from .models.file import FileUpload, FileAnalysisResult

__all__ = [
    # 数据库连接
    'get_db',
    'engine', 
    'AsyncSessionLocal',
    
    # 模型基类
    'Base',
    
    # 数据模型
    'ScanTask',
    'ScanResult', 
    'ScanRecord',
    'AIAnalysisResult',
    # 'FileUpload',
    # 'FileAnalysisResult',
]
